/*************************************************
 Copyright (c) 2017 
 Copyright (c) 2017 bruce <3350207067@qq.com>

 Author:    bruce
 Version:   1.0
 Date:      2017-10-19
 Description:
 
 ************************************************/
 

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "ff_subtitle_def.h"

#include "ijkplayer.h"

/*******************************************************************

*******************************************************************/
static void RotateL(char *In, int len, int loop)
{
    char szTmp[256] = {0};

    if (len >= 256) return;
    if (loop==0 || loop>=256) return;

    memset(szTmp, 0x00, sizeof(szTmp));

    memcpy(szTmp, In, loop);
    memmove(In, In+loop, len-loop);
    memcpy(In+len-loop, szTmp, loop);
}

/*******************************************************************

*******************************************************************/
static void ByteToBit(char *Out, const char *In, int bits)
{
    int i;
    for (i=0; i<bits; ++i)
        Out[i] = (In[i>>3]>>(7 - (i&7))) & 1;
}

/*******************************************************************

*******************************************************************/
const char LOOP_Table[16] =
{
    1, 1, 2, 2, 2, 2, 2, 2, 1, 2, 2, 2, 2, 2, 2, 1
};

const char PC1_Table[56] =
{
    57, 49, 41, 33, 25, 17,  9,
    1, 58, 50, 42, 34, 26, 18,
    10,  2, 59, 51, 43, 35, 27,
    19, 11,  3, 60, 52, 44, 36,
    63, 55, 47, 39, 31, 23, 15,
    7, 62, 54, 46, 38, 30, 22,
    14,  6, 61, 53, 45, 37, 29,
    21, 13,  5, 28, 20, 12,  4
};

const char PC2_Table[48] =
{
    14, 17, 11, 24,  1,  5,
    3, 28, 15,  6, 21, 10,
    23, 19, 12,  4, 26,  8,
    16,  7, 27, 20, 13,  2,
    41, 52, 31, 37, 47, 55,
    30, 40, 51, 45, 33, 48,
    44, 49, 39, 56, 34, 53,
    46, 42, 50, 36, 29, 32
};

/*******************************************************************

*******************************************************************/
static void Transform(char *Out, const char *In, const char *Table, int len)
{
    char szTmp[256] = {0};
    int i;

    if (!Out || !In || !Table) return;
    if (len >= 256) return;

    memset(szTmp, 0x00, sizeof(szTmp));
    for (i=0; i<len; ++i) szTmp[i] = In[Table[i]-1];

    memcpy(Out, szTmp, len);
}

static void SetSubKey(PSubKey pSubKey, const char Key[8])
{
    int i;
    char K[64] = {0}, *KL=&K[0], *KR=&K[28];

    ByteToBit(K, Key, 64);

    Transform(K, K, PC1_Table, 56);

    for ( i=0; i<16; ++i)
    {
        RotateL(KL, 28, LOOP_Table[i]);
        RotateL(KR, 28, LOOP_Table[i]);
        Transform((*pSubKey)[i], K, PC2_Table, 48);
    }
}

const char IP_Table[64] =
{
    58, 50, 42, 34, 26, 18, 10, 2,
    60, 52, 44, 36, 28, 20, 12, 4,
    62, 54, 46, 38, 30, 22, 14, 6,
    64, 56, 48, 40, 32, 24, 16, 8,
    57, 49, 41, 33, 25, 17,  9, 1,
    59, 51, 43, 35, 27, 19, 11, 3,
    61, 53, 45, 37, 29, 21, 13, 5,
    63, 55, 47, 39, 31, 23, 15, 7
};

const char E_Table[48] =
{
    32,  1,  2,  3,  4,  5,
    4,  5,  6,  7,  8,  9,
    8,  9, 10, 11, 12, 13,
    12, 13, 14, 15, 16, 17,
    16, 17, 18, 19, 20, 21,
    20, 21, 22, 23, 24, 25,
    24, 25, 26, 27, 28, 29,
    28, 29, 30, 31, 32,  1
};

const char P_Table[32] =
{
    16, 7, 20, 21,
    29, 12, 28, 17,
    1,  15, 23, 26,
    5,  18, 31, 10,
    2,  8, 24, 14,
    32, 27, 3,  9,
    19, 13, 30, 6,
    22, 11, 4,  25
};

/*******************************************************************

*******************************************************************/
static void Xor(char *InA, const char *InB, int len)
{
    int i;
    for (i=0; i<len; ++i) InA[i] ^= InB[i];
}

const char S_Box[8][4][16] =
{
    {
        {14,  4, 13,  1,  2, 15, 11,  8,  3, 10,  6, 12,  5,  9,  0,  7},
        { 0, 15,  7,  4, 14,  2, 13,  1, 10,  6, 12, 11,  9,  5,  3,  8},
        { 4,  1, 14,  8, 13,  6,  2, 11, 15, 12,  9,  7,  3, 10,  5,  0},
        {15, 12,  8,  2,  4,  9,  1,  7,  5, 11,  3, 14, 10,  0,  6, 13}
    },

    {
        {15,  1,  8, 14,  6, 11,  3,  4,  9,  7,  2, 13, 12,  0,  5, 10},
        { 3, 13,  4,  7, 15,  2,  8, 14, 12,  0,  1, 10,  6,  9, 11,  5},
        { 0, 14,  7, 11, 10,  4, 13,  1,  5,  8, 12,  6,  9,  3,  2, 15},
        {13,  8, 10,  1,  3, 15,  4,  2, 11,  6,  7, 12,  0,  5, 14,  9}
    },

    {
        {10,  0,  9, 14,  6,  3, 15,  5,  1, 13, 12,  7, 11,  4,  2,  8},
        {13,  7,  0,  9,  3,  4,  6, 10,  2,  8,  5, 14, 12, 11, 15,  1},
        {13,  6,  4,  9,  8, 15,  3,  0, 11,  1,  2, 12,  5, 10, 14,  7},
        { 1, 10, 13,  0,  6,  9,  8,  7,  4, 15, 14,  3, 11,  5,  2, 12}
    },

    {
        { 7, 13, 14,  3,  0,  6,  9, 10,  1,  2,  8,  5, 11, 12,  4, 15},
        {13,  8, 11,  5,  6, 15,  0,  3,  4,  7,  2, 12,  1, 10, 14,  9},
        {10,  6,  9,  0, 12, 11,  7, 13, 15,  1,  3, 14,  5,  2,  8,  4},
        { 3, 15,  0,  6, 10,  1, 13,  8,  9,  4,  5, 11, 12,  7,  2, 14}
    },

    {
        { 2, 12,  4,  1,  7, 10, 11,  6,  8,  5,  3, 15, 13,  0, 14,  9},
        {14, 11,  2, 12,  4,  7, 13,  1,  5,  0, 15, 10,  3,  9,  8,  6},
        { 4,  2,  1, 11, 10, 13,  7,  8, 15,  9, 12,  5,  6,  3,  0, 14},
        {11,  8, 12,  7,  1, 14,  2, 13,  6, 15,  0,  9, 10,  4,  5,  3}
    },

    {
        {12,  1, 10, 15,  9,  2,  6,  8,  0, 13,  3,  4, 14,  7,  5, 11},
        {10, 15,  4,  2,  7, 12,  9,  5,  6,  1, 13, 14,  0, 11,  3,  8},
        { 9, 14, 15,  5,  2,  8, 12,  3,  7,  0,  4, 10,  1, 13, 11,  6},
        { 4,  3,  2, 12,  9,  5, 15, 10, 11, 14,  1,  7,  6,  0,  8, 13}
    },

    {
        { 4, 11,  2, 14, 15,  0,  8, 13,  3, 12,  9,  7,  5, 10,  6,  1},
        {13,  0, 11,  7,  4,  9,  1, 10, 14,  3,  5, 12,  2, 15,  8,  6},
        { 1,  4, 11, 13, 12,  3,  7, 14, 10, 15,  6,  8,  0,  5,  9,  2},
        { 6, 11, 13,  8,  1,  4, 10,  7,  9,  5,  0, 15, 14,  2,  3, 12}
    },

    {
        {13,  2,  8,  4,  6, 15, 11,  1, 10,  9,  3, 14,  5,  0, 12,  7},
        { 1, 15, 13,  8, 10,  3,  7,  4, 12,  5,  6, 11,  0, 14,  9,  2},
        { 7, 11,  4,  1,  9, 12, 14,  2,  0,  6, 10, 13, 15,  3,  5,  8},
        { 2,  1, 14,  7,  4, 10,  8, 13, 15, 12,  9,  0,  3,  5,  6, 11}
    }
};

/*******************************************************************

*******************************************************************/
static void S_func(char Out[32], const char In[48])
{
    int i,j,k,l;
    for (i=0,j=0,k=0; i<8; ++i,In+=6,Out+=4)
    {
        j = (In[0]<<1) + In[5];
        k = (In[1]<<3) + (In[2]<<2) + (In[3]<<1) + In[4];

        for ( l=0; l<4; ++l)
            Out[l] = (S_Box[i][j][k]>>(3 - l)) & 1;
    }
}

/*******************************************************************

*******************************************************************/
static void F_func(char In[32], const char Ki[48])
{
    char MR[48] = {0};
    memset(MR, 0x00, sizeof(MR));

    Transform(MR, In, E_Table, 48);
    Xor(MR, Ki, 48);
    S_func(In, MR);
    Transform(In, In, P_Table, 32);
}

const char IPR_Table[64] =
{
    40, 8, 48, 16, 56, 24, 64, 32,
    39, 7, 47, 15, 55, 23, 63, 31,
    38, 6, 46, 14, 54, 22, 62, 30,
    37, 5, 45, 13, 53, 21, 61, 29,
    36, 4, 44, 12, 52, 20, 60, 28,
    35, 3, 43, 11, 51, 19, 59, 27,
    34, 2, 42, 10, 50, 18, 58, 26,
    33, 1, 41,  9, 49, 17, 57, 25
};

/*******************************************************************

*******************************************************************/
static void BitToByte(char *Out, const char *In, int bits)
{
    int i;
    memset(Out, 0, bits>>3);
    for (i=0; i<bits; ++i) Out[i>>3] |= In[i]<<(7 - (i&7));
}

/*******************************************************************

*******************************************************************/
static void DES(char Out[8], const char In[8], const PSubKey pSubKey, int Type)
{
    int i;
    char M[64] = {0}, *ML=&M[0], *MR=&M[32], szTmp[32] = {0};

    ByteToBit(M, In, 64);
    Transform(M, M, IP_Table, 64);

    if (Type == ENCRYPT)
    {
        for (i=0; i<16; ++i)
        {
            memcpy(szTmp, MR, 32);
            F_func(MR, (*pSubKey)[i]);
            Xor(MR, ML, 32);
            memcpy(ML, szTmp, 32);
        }
    }
    else
    {
        for (i=15; i>=0; --i)
        {
            memcpy(szTmp, MR, 32);
            F_func(MR, (*pSubKey)[i]);
            Xor(MR, ML, 32);
            memcpy(ML, szTmp, 32);
        }
    }
    RotateL(M, 64, 32);
    Transform(M, M, IPR_Table, 64);
    BitToByte(Out, M, 64);
}

/*******************************************************************

*******************************************************************/
int Run3Des(int bType, int bMode, const char *In, unsigned int in_len, 
    const char *Key, unsigned int key_len, char* Out, unsigned int out_len, 
    const char cvecstr[8])
{
    int i,j,k;
    char m_SubKey[3][16][48];
    unsigned char nKey;

    if (!In || !Key || !Out) return 0;

    if (in_len & 0x00000007) return 0;

    if (key_len & 0x00000007) return 0;

    if (out_len < in_len) return 0;

    nKey = (key_len>>3)>3 ? 3 : (key_len>>3);
    memset(m_SubKey, 0x00, sizeof(m_SubKey));
    for ( i=0; i<nKey; i++)
    {
        SetSubKey(&m_SubKey[i], &Key[i<<3]);
    }

    if (bMode == ECB)
    {
        if (nKey == 1)
        {
            for (i=0,j=in_len>>3; i<j; ++i,Out+=8,In+=8)
            {
                DES(Out, In, &m_SubKey[0], bType);
            }
        }
        else if (nKey == 2)
        {
            for (i=0,j=in_len>>3; i<j; ++i,Out+=8,In+=8)
            {
                DES(Out, In,  &m_SubKey[0], bType);
                DES(Out, Out, &m_SubKey[1], bType==ENCRYPT?DECRYPT:ENCRYPT);
                DES(Out, Out, &m_SubKey[0], bType);
            }
        }
        else if (nKey == 3)
        {
            for (i=0,j=in_len>>3; i<j; ++i,Out+=8,In+=8)
            {
                DES(Out, In,  &m_SubKey[bType?2:0], bType);
                DES(Out, Out, &m_SubKey[1],         bType==ENCRYPT?DECRYPT:ENCRYPT);
                DES(Out, Out, &m_SubKey[bType?0:2], bType);
            }
        }
        else
        {
            return 0;
        }
    }
    else if (bMode == CBC)
    {
        if (cvecstr == NULL) return 0;

        char cvec[8] = {0};
        char cvin[8] = {0};

        memcpy(cvec, cvecstr, 8);

        if (nKey == 1)
        {
            for (i=0,j=in_len>>3; i<j; ++i,Out+=8,In+=8)
            {
                if (bType == ENCRYPT)
                {
                    for (k=0; k<8; ++k)
                    {
                        cvin[k] = In[k] ^ cvec[k];
                    }
                }
                else
                {
                    memcpy(cvin, In, 8);
                }

                DES(Out, cvin, &m_SubKey[0], bType);

                if (bType == ENCRYPT)
                {
                    memcpy(cvec, Out, 8);
                }
                else
                {
                    for (k=0; k<8; ++k)
                    {
                        Out[k] = Out[k] ^ cvec[k];
                    }
                    memcpy(cvec, cvin, 8);
                }
            }
        }
        else if (nKey == 2)
        {
            for (i=0,j=in_len>>3; i<j; ++i,Out+=8,In+=8)
            {
                if (bType == ENCRYPT)
                {
                    for ( k=0; k<8; ++k)
                    {
                        cvin[k] = In[k] ^ cvec[k];
                    }
                }
                else
                {
                    memcpy(cvin, In, 8);
                }

                DES(Out, cvin, &m_SubKey[0], bType);
                DES(Out, Out,  &m_SubKey[1], bType==ENCRYPT?DECRYPT:ENCRYPT);
                DES(Out, Out,  &m_SubKey[0], bType);

                if (bType == ENCRYPT)
                {
                    memcpy(cvec, Out, 8);
                }
                else
                {
                    for (k=0; k<8; ++k)
                    {
                        Out[k] = Out[k] ^ cvec[k];
                    }
                    memcpy(cvec, cvin, 8);
                }
            }
        }
        else if (nKey == 3)
        {

            for (i=0,j=in_len>>3; i<j; ++i,Out+=8,In+=8)
            {
                if (bType == ENCRYPT)
                {
                    for (k=0; k<8; ++k)
                    {
                        cvin[k] = In[k] ^ cvec[k];
                    }
                }
                else
                {
                    memcpy(cvin, In, 8);
                }

                DES(Out, cvin, &m_SubKey[bType?2:0], bType);
                DES(Out, Out,  &m_SubKey[1],         bType==ENCRYPT?DECRYPT:ENCRYPT);
                DES(Out, Out,  &m_SubKey[bType?0:2], bType);

                if (bType == ENCRYPT)
                {
                    memcpy(cvec, Out, 8);
                }
                else
                {
                    for (k=0; k<8; ++k)
                    {
                        Out[k] = Out[k] ^ cvec[k];
                    }
                    memcpy(cvec, cvin, 8);
                }
            }
        }
        else
        {
            return 0;
        }
    }
    else
    {
        return 0;
    }

    return 1;
}

int RunRsm(char *pText)
{
    int len;
    int pad;

    len=strlen(pText);
    pad=*(pText+len-1);
    DBG_SUBTITLE("bruce >>>>>> pad:%d, len:%d", pad, len);
    *(pText+len-pad)=0x00;
    return pad;
}

const char * base64char = 
    "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";

int base64_decode(const char *base64, unsigned char *bindata)
{
    int i, j;
    unsigned char k;
    unsigned char temp[4];
    for ( i = 0, j = 0; base64[i] != '\0' ; i += 4 )
    {
        memset( temp, 0xFF, sizeof(temp) );
        for ( k = 0 ; k < 64 ; k ++ )
        {
            if ( base64char[k] == base64[i])
                temp[0]= k;
        }
        for ( k = 0 ; k < 64 ; k ++ )
        {
            if ( base64char[k] == base64[i+1])
                temp[1]= k;
        }
        for ( k = 0 ; k < 64 ; k ++ )
        {
            if ( base64char[k] == base64[i+2])
                temp[2]= k;
        }
        for ( k = 0 ; k < 64 ; k ++ )
        {
            if (base64char[k] == base64[i+3])
                temp[3]= k;
        }

        bindata[j++] = ((unsigned char)(((unsigned char)(temp[0] << 2))&0xFC)) |
                ((unsigned char)((unsigned char)(temp[1]>>4)&0x03));
        if ( base64[i+2] == '=' )
            break;

        bindata[j++] = ((unsigned char)(((unsigned char)(temp[1] << 4))&0xF0)) |
                ((unsigned char)((unsigned char)(temp[2]>>2)&0x0F));
        if ( base64[i+3] == '=' )
            break;

        bindata[j++] = ((unsigned char)(((unsigned char)(temp[2] << 6))&0xF0)) |
                ((unsigned char)(temp[3]&0x3F));
    }
    return j;
}

int ffp_3Des(int bType, int bMode, const char *in, char *out, int binLen)
{
    int ret;    
    char iv[8] = {0};
    const char *key = "a8Uh5Se8WryPBdKFwdpF6LfP";
    
    memset(iv, 0, sizeof(iv));
    ret = Run3Des(bType, bMode, in, binLen, key, strlen(key), out, binLen, iv);
	if (ret != 1)
	{
		return ERROR;
	}

	(void)RunRsm(out);
	return OK;
}
